package com.indi.ssoclient.controller;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.client.RestTemplate;

import javax.servlet.http.HttpSession;
import java.util.ArrayList;
import java.util.List;

@Controller
public class HelloController {
    @Value("${sso.server.url}")
    public String ssoServerUrl;

    /**
     * 无需登录就可访问
     *
     * @return
     */
    public String hello() {
        return "hello";
    }

    /**
     * 登录之后，才能访问的页面
     *
     * @param model
     * @param session
     * @param token   从ssoserver登录成功之后，重定向回来会携带这个参数
     * @return
     */
    @GetMapping("/boss")
    public String employeess(Model model, HttpSession session,
                             @RequestParam(value = "token", required = false) String token) {
        if (!StringUtils.isEmpty(token)) {
            // 这说明是在ssoserver登录成功跳回来的
            /*
                因为登录服务器，只返回了一个token，所以我们没办法只存一个token到session中，
                TODO 所以还需要去登录服务器，请求到token对应的用户数据
             */
            // 这里做了简单处理
            RestTemplate restTemplate = new RestTemplate();
            ResponseEntity<String> forEntity = restTemplate.getForEntity("http://ssoserver.com:8080/user-info?token=" + token, String.class);
            String body = forEntity.getBody();
            session.setAttribute("loginUser", body);
        }

        Object attribute = session.getAttribute("loginUser");
        if (attribute == null) {
            // 没登录，跳转到登录服务器进行登录
            // 跳转过去之后，使用url上的查询参数，表示自己是哪个界面
            return "redirect:" + ssoServerUrl + "?redirectUrl=http://client2.com:8082/boss";
        } else {
            List<String> emps = new ArrayList<>();
            emps.add("张三");
            emps.add("李四");
            model.addAttribute("emps", emps);
            return "list";

        }
    }
}
